Skip to content

[skills] chore: rename overlap and recompute skills with nemo mbridge prefix - #4031

Merged
cuichenx merged 7 commits into
mainfrom
chore/sign-nemo-mbridge-batch-04
May 29, 2026
Merged

[skills] chore: rename overlap and recompute skills with nemo mbridge prefix#4031
cuichenx merged 7 commits into
mainfrom
chore/sign-nemo-mbridge-batch-04

Conversation

@cuichenx

Copy link
Copy Markdown
Contributor

Summary

  • Rename this batch of public skills to the nemo-mbridge-* prefix:
    • nemo-mbridge-perf-expert-parallel-overlap
    • nemo-mbridge-perf-moe-comm-overlap
    • nemo-mbridge-perf-activation-recompute
  • Add/keep license: Apache-2.0 frontmatter for these public skills.
  • Add placeholder evals/evals.json files where needed for NVSkills signing.
  • Update docs/skills-index.md references for the renamed skills.

Validation

  • Verified selected public skill directory names match SKILL.md frontmatter names.
  • Verified selected public skills have license: Apache-2.0.
  • Verified selected public skills have valid placeholder evals/evals.json files.
  • Verified selected public root SKILL.md files are <= 500 lines.
  • git diff --check passed.
  • uv run --no-sync pre-commit run --all-files passed.

@cuichenx

Copy link
Copy Markdown
Contributor Author

/nvskills-ci

1. **Set `PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True` first** — most
borderline OOMs are caused by memory fragmentation, not capacity. This
fixes it at zero cost. See @skills/perf-memory-tuning/SKILL.md.
fixes it at zero cost. See @skills/mbridge-perf-memory-tuning/SKILL.md.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Bug: This references @skills/mbridge-perf-memory-tuning/SKILL.md, but the actual directory is still skills/perf-memory-tuning/. This name was not renamed in this PR (and mbridge-perf-memory-tuning doesn't match the nemo-mbridge-* prefix pattern used elsewhere in this PR either). This is a broken cross-reference.

Suggested change
fixes it at zero cost. See @skills/mbridge-perf-memory-tuning/SKILL.md.
fixes it at zero cost. See @skills/perf-memory-tuning/SKILL.md.

- Combining `mlp` + `core_attn` is slightly worse than `mlp` alone
- For this workload, the actual OOM fix was `PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True`
(memory fragmentation, not capacity). See @skills/perf-memory-tuning/SKILL.md.
(memory fragmentation, not capacity). See @skills/mbridge-perf-memory-tuning/SKILL.md.

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same broken reference — mbridge-perf-memory-tuning doesn't exist.

Suggested change
(memory fragmentation, not capacity). See @skills/mbridge-perf-memory-tuning/SKILL.md.
(memory fragmentation, not capacity). See @skills/perf-memory-tuning/SKILL.md.

if_still_oom: "add layernorm (cheap) or mlp (expensive but saves ~3 GB)"
last_resort: "recompute_granularity=full, recompute_method=uniform"
alternative: "see skills/perf-memory-tuning/ for VPP tuning and other memory strategies"
alternative: "see skills/mbridge-perf-memory-tuning/ for VPP tuning and other memory strategies"

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Same issue — skills/mbridge-perf-memory-tuning/ doesn't exist. Should remain skills/perf-memory-tuning/ until that skill is actually renamed.

Suggested change
alternative: "see skills/mbridge-perf-memory-tuning/ for VPP tuning and other memory strategies"
alternative: "see skills/perf-memory-tuning/ for VPP tuning and other memory strategies"

@claude

claude Bot commented May 28, 2026

Copy link
Copy Markdown
Contributor

Review: Broken cross-references to non-existent skill directory -- Three places in the renamed nemo-mbridge-perf-activation-recompute skill now reference skills/mbridge-perf-memory-tuning/SKILL.md (SKILL.md lines 32, 132; card.yaml line 95). The actual directory is still skills/perf-memory-tuning/ -- it was not renamed in this PR, and the target name mbridge-perf-memory-tuning does not match the nemo-mbridge-* prefix convention used for the other renames. These are broken cross-references. See inline comments for suggested fixes. -- Stale references in files not touched by this PR: Per CONTRIBUTING.md, renames must update all references. Files still using old names: skills/perf-memory-tuning/SKILL.md (3 refs to perf-activation-recompute), skills/perf-cuda-graphs/SKILL.md and card.yaml (2 refs to perf-activation-recompute), docs/training/communication-overlap.md (3 refs to perf-expert-parallel-overlap, 2 refs to perf-moe-comm-overlap), docs/training/activation-recomputation.md (1 ref to perf-activation-recompute), docs/training/moe-optimization.md (1 ref to perf-moe-comm-overlap). These should be updated. -- Suggested test cases: No perf tests impacted.

@yaoyu-33 yaoyu-33 added area:misc Cross-cutting utilities, logging, helpers, and other changes docs Documentation-only updates or documentation debt needs-review PR is ready for code review and waiting on a reviewer labels May 28, 2026
Signed-off-by: nvskills-svc-account <svc-nvskills-signing@nvidia.com>
@copy-pr-bot

copy-pr-bot Bot commented May 28, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

Signed-off-by: Chen Cui <chcui@nvidia.com>
@cuichenx

Copy link
Copy Markdown
Contributor Author

/nvskills-ci

@cuichenx cuichenx added the docs-only With great power comes great responsibility. label May 28, 2026
Signed-off-by: Chen Cui <chcui@nvidia.com>
@cuichenx

Copy link
Copy Markdown
Contributor Author

Scoped this signing PR back down to the renamed skill directories only. The cross-skill reference updates that touched old skill directories were moved to draft cleanup PR #4044 so NVSkills should not validate unrelated old skill dirs in this batch.

@cuichenx

Copy link
Copy Markdown
Contributor Author

/nvskills-ci

Signed-off-by: Chen Cui <chcui@nvidia.com>
@cuichenx

Copy link
Copy Markdown
Contributor Author

Kept the activation-recompute references pointed at the final nemo-mbridge memory-tuning skill name so this batch does not pull the old memory-tuning skill into validation. Rerunning NVSkills now.

@cuichenx

Copy link
Copy Markdown
Contributor Author

/nvskills-ci

Signed-off-by: nvskills-svc-account <svc-nvskills-signing@nvidia.com>
@cuichenx

Copy link
Copy Markdown
Contributor Author

/ok to test 1921fa5

Signed-off-by: Chen Cui <chcui@nvidia.com>
@cuichenx
cuichenx merged commit 1c61a72 into main May 29, 2026
3 checks passed
@cuichenx
cuichenx deleted the chore/sign-nemo-mbridge-batch-04 branch May 29, 2026 16:23
@Phlip79 Phlip79 mentioned this pull request May 29, 2026
3 tasks
vasunvidia pushed a commit to vasunvidia/Megatron-Bridge that referenced this pull request Jun 10, 2026
… prefix (NVIDIA-NeMo#4031)

Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: nvskills-svc-account <svc-nvskills-signing@nvidia.com>
Co-authored-by: nvskills-svc-account <svc-nvskills-signing@nvidia.com>
Signed-off-by: Vasudevan Rengasamy <vrengasamy@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:misc Cross-cutting utilities, logging, helpers, and other changes docs Documentation-only updates or documentation debt docs-only With great power comes great responsibility. needs-review PR is ready for code review and waiting on a reviewer

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants